from _md5 import md5

import requests
from lxml import etree

# 导超级鹰识别验证码接口
class Chaojiying_Client(object):
    def __init__(self, username, password, soft_id):
        self.username = username
        password =  password.encode('utf8')
        self.password = md5(password).hexdigest()
        self.soft_id = soft_id
        self.base_params = {
            'user': self.username,
            'pass2': self.password,
            'softid': self.soft_id,
        }
        self.headers = {
            'Connection': 'Keep-Alive',
            'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
        }

    def PostPic(self, im, codetype):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
        }
        params.update(self.base_params)
        files = {'userfile': ('ccc.jpg', im)}
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files, headers=self.headers)
        return r.json()

    def ReportError(self, im_id):
        """
        im_id:报错题目的图片ID
        """
        params = {
            'id': im_id,
        }
        params.update(self.base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.headers)
        return r.json()

if __name__ == '__main__':
    url = 'http://www.renren.com/SysHome.do'
    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/65.0.3325.181 Safari/537.36'
    }
    page_text = requests.get(url=url, headers=headers).text
    tree = etree.HTML(page_text)
    code_img_src = tree.xpath('//*[@id="verifyPic_login"]/@src')[0]
    code_img_data =  requests.get(url=code_img_src, headers=headers).content
    with open('./code.jpg', 'wb') as fp:
        fp.write(code_img_data)

    # 使用超级鹰提供的示例代码对验证码图片进行识别
    chaojiying = Chaojiying_Client('whxchaojiying', '20210405QED@', '914815')
    code_path = './code.jpg'
    im = open(code_path, 'rb').read()
    code_result = chaojiying.PostPic(im, 1006)

    # post请求的发送 (模拟登录)
    login_url = 'http://www.renren.com/ajaxLogin/login?1=1&uniqueTimestamp=20213161516'
    data = {
        'email':'15615723073',
        'icode': code_result['pic_str'],
        'origURL':'http://www.renren.com/home',
        'domain':'renren.com',
        'key_id':'1',
        'captcha_type':'web_login',
        'password':'409252a9f2a94daf5a2ae88f17a5016f1d1d135729bbf83137329b42882faa72',
        'rkey':'88c3ce72a86f84868e5006432a0566bc',
        'f':'http%3A%2F%2Fwww.renren.com%2F976550806%2Fnewsfeed%2Fphoto'
    }
    print(code_result['pic_str'])
    response = requests.post(url=login_url, data=data, headers=headers)
    print(response.status_code)
    # login_page_text = response.text
    # with open('renren.html', 'w', encoding='utf-8') as fp:
    #     fp.write(login_page_text)